import random

import tool.tool
from BinaryTree import BinaryTree
from analysis.generator import generate_tree, set_random_leaves, generateMatrix, generateRandomNumberList


def generate_data(n):
    match n:
        case 1:
            node = generate_tree(0, 3, 1, 9)
            node = set_random_leaves(node, 3, 0)
            return BinaryTree(node)
        case 2:
            return generateMatrix()
        case 3:
            return
        case 4:
            return random.randint(100, 200)
        case 5:
            return generateRandomNumberList(amount=5,min_num=1, max_num=200)

